package com.fan.leetcode;

import org.junit.jupiter.api.Test;

import java.util.Arrays;
import java.util.stream.Collectors;

/**
 * 面试题 17.14. 最小K个数
 *
 * @author ZhangFan
 * @since 2021/9/3 下午 10:04
 */
public class QuestionInterview17_14 {
    @Test
    void test() {
        smallestK(new int[]{1,3,5,7,2,4,6,8}, 4);
    }

    /**
     * 暴力解题
     *
     * @param arr 数组
     * @param k   最小的k个数
     * @return 最小k个数的数组
     */
    /*public int[] smallestK(int[] arr, int k) {
        Arrays.sort(arr);
        int[] ints = Arrays.stream(arr).limit(k).toArray();
        return ints;
    }*/
    public int[] smallestK(int[] arr, int k) {
        int[] vec = new int[k];
        Arrays.sort(arr);
        for (int i = 0; i < k; ++i) {
            vec[i] = arr[i];
        }
        return vec;
    }
}
